export MODEL=Qwen2.5-Coder-3B-Instruct-Debug
export DS_TYPE=primary
#export DS_TYPE=plus
#export PROMPT_TYPE=zero
export PROMPT_TYPE=cot
echo "MODEL=$MODEL, PROMPT_TYPE=$PROMPT_TYPE, DS_TYPE=$DS_TYPE"
python vllm_inference.py \
    --base_model $MODEL \
    --dataset "debug" \
    --input_data_dir "./data/" \
    --output_data_dir "./judge/solution_folder/" \
    --batch_size 1 \
    --num_of_sequences 1 \
	--ds_type $DS_TYPE \
    --prompt_type $PROMPT_TYPE \
    --start_idx 0 \
    --end_idx -1
